# -*- coding: utf-8 -*-
# File : db_init.py
# Author: taoyahui
# Date : 2022/5/9

# 初始化数据
import sqlite3
print(f"{'*' * 20} 初始化开始，创建student表 {'*' * 20}")

# 第一步，创建数据库连接
connect = sqlite3.connect("students.db")

# 第二步, 创建connect变量对应的游标
cursor = connect.cursor()

# 第三步，执行数据库操作
# 操作1，判断数据库中是否有student表，如果有的话就对表进行删除
cursor.execute("DROP TABLE IF EXISTS student")
# 操作2, 创建student表
cursor.execute('''CREATE TABLE student(
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name VARCHAR(20),
            gender CHAR(1),
            student_id CHAR(8),
            class_name VARCHAR(20),
            college_name VARCHAR(20)
        )''')

# 第四步，执行创建student表的操作
connect.commit()

# 第五步，执行向student表中插入数据
cursor.execute("INSERT INTO student VALUES(null, '张三', '男', '20220001', '区块链211', '软大学院')")
cursor.execute("INSERT INTO student VALUES(null, '李四', '男', '20220002', '区块链211', '软大学院')")
cursor.execute("INSERT INTO student VALUES(null, '王五', '男', '20220003', '区块链211', '软大学院')")
cursor.execute("INSERT INTO student VALUES(null, '赵六', '男', '20220004', '区块链211', '软大学院')")

# 第六步，执行student表插入操作对应的提交
connect.commit()

# 到这里就完成了初始化中关于数据库的相关操作（建表，插入数据）
# 第七步，关闭连接
cursor.close()
connect.close()

print(f"{'*' * 20} 初始化操作已完成 {'*' * 20}")




